# -*- coding:utf-8 -*-
# Author:凌逆战 | Never
# Date: 2023/6/29
"""

"""
import numpy as np
import torch
from torch_audiomentations import Compose, Gain, PolarityInversion, HighPassFilter

samples = np.array(
    [
        [[0.75, 0.5, -0.25, -0.125, 0.0], [0.65, 0.5, -0.25, -0.125, 0.0]],
        [[0.3, 0.5, -0.25, -0.125, 0.0], [0.9, 0.5, -0.25, -0.125, 0.0]],
        [[0.9, 0.5, -0.25, -1.06, 0.0], [0.9, 0.5, -0.25, -1.12, 0.0]],
    ],
    dtype=np.float32,
)
sample_rate = 16000

augment = HighPassFilter(p=1.0, output_type="dict")
processed_samples = augment(
    samples=torch.from_numpy(samples), sample_rate=sample_rate
).samples.numpy()


# self.assertEqual(processed_samples.shape, samples.shape)
# self.assertEqual(processed_samples.dtype, np.float32)
